#include "ota_http.h"
#include "log/log.h"
#include <curl/curl.h>
#include <stdlib.h>
#include <string.h>
#include <stdio.h>


size_t write_callback(void *ptr, size_t size, size_t nmemb, void *userdata)
{
    char *json_buf = (char *)userdata;
    // 数据长度
    int data_len = size * nmemb;
    // 将数据学入容器
    memcpy(json_buf, ptr, data_len);
    // 添加结束符
    json_buf[data_len] = '\0';
    // 返回数据长度
    return data_len;
}

/**
 * 请求获取json数据
 */
char *ota_http_getJSON(char *url)
{
    // 初始化curl
    CURL *curl = curl_easy_init();
    // 指定url
    curl_easy_setopt(curl, CURLOPT_URL, url);
    // 指定接收响应数据的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, write_callback);
    // 指定接收响应数据的容器，传递给回调函数，收到数据后，将数据写入该容器
    char *json_buf = malloc(1024);
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, json_buf);
    // 执行请求
    CURLcode res = curl_easy_perform(curl);
    // 如果失败，释放资源
    if (res != CURLE_OK)
    {
        log_error("curl_easy_perform() failed: %s\n", curl_easy_strerror(res));
        free(json_buf);
        curl_easy_cleanup(curl);
        return NULL;
    }
    // 请求成功释放curl
    curl_easy_cleanup(curl);

    return json_buf;
}

/**
 * 下载文件
 */
int ota_http_download(char *url , char *filename)
{
    // 初始化curl
    CURL *curl = curl_easy_init();
    // 指定url
    curl_easy_setopt(curl, CURLOPT_URL, url);
    
    // 指定接收响应数据的回调函数
    curl_easy_setopt(curl, CURLOPT_WRITEFUNCTION, fwrite);
    // 指定得到返回的文件的回调函数的参数，即文件指针
    FILE *fp = fopen(filename, "wb");
    curl_easy_setopt(curl, CURLOPT_WRITEDATA, fp);  
    // 执行请求
    CURLcode res = curl_easy_perform(curl);
    
    // 如果失败，释放资源
    if (res != CURLE_OK)
    {
        log_error("curl_easy_perform() failed: %s\n", curl_easy_strerror(res));
        fclose(fp);
        curl_easy_cleanup(curl);
        return -1;
    }
    // 请求成功释放curl
    fclose(fp);
    curl_easy_cleanup(curl);
    
    return 0;
}
